import numpy as np

a= 10*np.random.random((5,4))
print(a)
ind = a.argmax(axis=0)
print(ind)
datamax = a[ind,range(a.shape[1])]
print(datamax)